import warnings

warnings.filterwarnings("ignore")
from transformers import BertConfig, BertModel


def test_model():
    config = BertConfig()
    model = BertModel(config)
    print(config)
    print(model)


def test_save_model():
    model = BertModel.from_pretrained("bert-base-cased")
    model.save_pretrained("../output/bert-base-cased/")


def test_load_model():
    model = BertModel.from_pretrained("../output/bert-base-cased/")
    print(model)


if __name__ == '__main__':
    # test_save_model()
    test_load_model()
